absl-py==2.0.0
aiohttp==3.9.1
aiosignal==1.3.1
async-timeout==4.0.3
attrs==23.1.0
cached-property==1.5.2
certifi==2023.11.17
charset-normalizer==3.3.2
chex==0.1.7
datasets==2.15.0
diffusers==0.24.0
dill==0.3.7
dm-tree==0.1.8
etils==1.3.0
filelock==3.13.1
flax==0.7.2
frozenlist==1.4.0
fsspec==2023.10.0
huggingface-hub==0.19.4
idna==3.6
importlib-metadata==7.0.0
importlib-resources==6.1.1
jax==0.4.13
jaxlib==0.4.13
Jinja2==3.1.2
markdown-it-py==3.0.0
MarkupSafe==2.1.3
mdurl==0.1.2
ml-dtypes==0.2.0
mpmath==1.3.0
msgpack==1.0.7
multidict==6.0.4
multiprocess==0.70.15
nest-asyncio==1.5.8
networkx==3.1
numpy==1.24.4
nvidia-cublas-cu12==12.1.3.1
nvidia-cuda-cupti-cu12==12.1.105
nvidia-cuda-nvrtc-cu12==12.1.105
nvidia-cuda-runtime-cu12==12.1.105
nvidia-cudnn-cu12==8.9.2.26
nvidia-cufft-cu12==11.0.2.54
nvidia-curand-cu12==10.3.2.106
nvidia-cusolver-cu12==11.4.5.107
nvidia-cusparse-cu12==12.1.0.106
nvidia-nccl-cu12==2.18.1
nvidia-nvjitlink-cu12==12.3.101
nvidia-nvtx-cu12==12.1.105
opt-einsum==3.3.0
optax==0.1.7
orbax-checkpoint==0.2.3
packaging==23.2
pandas==2.0.3
Pillow==10.1.0
pyarrow==14.0.1
pyarrow-hotfix==0.6
Pygments==2.17.2
python-dateutil==2.8.2
pytz==2023.3.post1
PyYAML==6.0.1
regex==2023.10.3
requests==2.31.0
rich==13.7.0
safetensors==0.4.1
scipy==1.10.1
six==1.16.0
sympy==1.12
tensorstore==0.1.45
tokenizers==0.15.0
toolz==0.12.0
torch==2.1.1
torchvision==0.16.1
tqdm==4.66.1
transformers==4.36.0
triton==2.1.0
typing_extensions==4.9.0
tzdata==2023.3
urllib3==2.1.0
xxhash==3.4.1
yarl==1.9.4
zipp==3.17.0
